eb9bf8484d228a7ea2914b79902161dfeaf995bf,deeplearning4j-core/src/main/java/org/deeplearning4j/parallelism/MagicQueue.java,MagicQueue,poll,#number#TimeUnit#,258

Before Change


            if (numberOfBuckets > 1) {
                int deviceId = Nd4j.getAffinityManager().getDeviceForCurrentThread();
                return backingQueues.get(deviceId).poll(time, timeUnit);
            } else return backingQueues.get(0).poll(time, timeUnit);
        } else {
            DataSet ds = backingQueues.get(interleavedCounter.getAndIncrement()).poll(time, timeUnit);
            if (interleavedCounter.get() >= backingQueues.size())

After Change



                return ds;
            } else {
                DataSet ds = backingQueues.get(0).poll(time, timeUnit);

                if (ds != null)
                    cntGet.incrementAndGet();

                return ds;
            }
        } else {
            DataSet ds = backingQueues.get(interleavedCounter.getAndIncrement()).poll(time, timeUnit);